"""
python fujian1_group.py

./myenv/Scripts/python fujian1_group.py
"""

import pandas as pd
import os
import json

# 读取 CSV 文件，添加 encoding 参数
csv_path = os.path.join('fujian', 'fujian1.csv')
data = pd.read_csv(csv_path, encoding='gbk')

# 将“月份”列转换为日期格式
data['月份'] = pd.to_datetime(data['月份'])

# 按品类分组并排序
grouped_data = data.groupby('品类')

# 指定保存目录
output_dir = os.path.join('fujian', 'fujian1', 'group_output')
os.makedirs(output_dir, exist_ok=True)

# 遍历每个品类，按时间排序并生成 JSON 文件
for category, group in grouped_data:
    sorted_group = group.sort_values(by='月份')
    
    # 创建 JSON 格式的数据
    json_data = [
        {"date": row['月份'].strftime('%Y-%m-%d'), "inventory": row['库存量']}
        for _, row in sorted_group.iterrows()
    ]
    
    # 保存为 JSON 文件
    json_filename = os.path.join(output_dir, f"inventory_category{category}.json")
    with open(json_filename, 'w', encoding='utf-8') as json_file:
        json.dump(json_data, json_file, ensure_ascii=False, indent=4)

    print(f"Saved data for category {category} to {json_filename}")
